package com.chenjiajin.listener;


import com.alibaba.excel.context.AnalysisContext;
import com.alibaba.excel.event.AnalysisEventListener;
import com.alibaba.excel.exception.ExcelDataConvertException;
import com.alibaba.fastjson.JSON;
import com.chenjiajin.domain.DemoImport;
import com.chenjiajin.service.TestService;
import lombok.extern.slf4j.Slf4j;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;

/**
 * 每解析一行会回调invoke()方法。
 * 整个excel解析结束会执行doAfterAllAnalysed()方法
 */

//有个很重要的点   不能被spring管理,要每次读取excel都要new。
@Slf4j
public class ExcelExportDataListener extends AnalysisEventListener<DemoImport> {

    private TestService service;

    public ExcelExportDataListener(TestService service) {
        this.service = service;
    }

    //每次读取100条数据就进行保存操作   分页读取
    private static final int BATCH_COUNT = 100;
    //由于每次读都是新new ExcelExportDataListener的，所以这个list不会存在线程安全问题
    List<DemoImport> list = new ArrayList<>();


    @Override
    public void invoke(DemoImport data, AnalysisContext analysisContext) {
        log.info("解析到一条数据:{}", JSON.toJSONString(data));
        list.add(data);
        if (list.size() >= BATCH_COUNT) {
            saveData();
            // 存储完成清理 list
            list.clear();
        }
    }

    @Override
    public void doAfterAllAnalysed(AnalysisContext analysisContext) {
        // 这里也要保存数据，确保最后遗留的数据也存储到数据库
        saveData();
        log.info("所有数据解析完成！");
    }

    private void saveData() {
            log.info("{}条数据，开始存储数据库！", list.size());
            //保存数据
            list.forEach(System.err::println);
            //  装箱单-数据存库
            service.saveBatch(list);
            log.info("存储数据库成功！");
    }


    /**
     * @param headMap 传入excel头部(第一行数据)数据的index,name
     * 校验excel头部格式，必须完全匹配
     */
    List<String> teamList = new ArrayList<>();
    int x = 0;
    @Override
    public void invokeHeadMap(Map<Integer, String> headMap, AnalysisContext context) {
        x++;
        super.invokeHeadMap(headMap, context);
        System.out.println(headMap.size());
        if (x == 3 && headMap.size() != 20) {
            teamList.add("解析Excel出错,请传入正确模板的Excel");
        }
    }


    /**
     * 监听器实现这个方法就可以在读取数据的时候获取到异常信息
     */
    @Override
    public void onException(Exception exception, AnalysisContext context) {
        log.error("解析失败，但是继续解析下一行:{}", exception.getMessage());
        // 如果是某一个单元格的转换异常 能获取到具体行号
        // 如果要获取头的信息 配合invokeHeadMap使用
        if (exception instanceof ExcelDataConvertException) {
            ExcelDataConvertException e = (ExcelDataConvertException) exception;
            log.error("第{}行，第{}列解析异常", e.getRowIndex(), e.getColumnIndex());
        }
    }


}